#ifndef GPU_CONVOLUTION
#define GPU_CONVOLUTION

#include "GPU_Mat.h"

namespace gpu_tianqichen
{
#ifdef __NOT_USE_GPU__
    	void convolution_v2h(
		Mat3D& DstH,
		const Mat3D& SrcV,
        const Mat4DBias& SrcMask){}

	void convolution_h2v(
		Mat3D& DstV,
		const Mat3D& SrcH,
        const Mat4DBias& SrcMask){}

	void convolution_a2m(
		Mat4DBias& DstMask,
		const Mat3D& SrcV,
        const Mat3D& SrcH ){}
#else
	void convolution_v2h(
		Mat3D& DstH,
		const Mat3D& SrcV,
		const Mat4DBias& SrcMask);

	void convolution_h2v(
		Mat3D& DstV,
		const Mat3D& SrcH,
		const Mat4DBias& SrcMask);

	void convolution_a2m(
		Mat4DBias& DstMask,
		const Mat3D& SrcV,
		const Mat3D& SrcH );
#endif
}

#endif

